// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "ipc/ipc_test_channel_listener.h"

#include "ipc/ipc_message.h"
#include "ipc/ipc_sender.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace IPC {

// static
void TestChannelListener::SendOneMessage(IPC::Sender* sender,
    const char* text)
{
    static int message_index = 0;

    IPC::Message* message = new IPC::Message(0,
        2,
        IPC::Message::PRIORITY_NORMAL);
    message->WriteInt(message_index++);
    message->WriteString(std::string(text));

    // Make sure we can handle large messages.
    char junk[kLongMessageStringNumBytes];
    memset(junk, 'a', sizeof(junk) - 1);
    junk[sizeof(junk) - 1] = 0;
    message->WriteString(std::string(junk));

    sender->Send(message);
}

bool TestChannelListener::OnMessageReceived(const IPC::Message& message)
{
    base::PickleIterator iter(message);

    int ignored;
    EXPECT_TRUE(iter.ReadInt(&ignored));
    std::string data;
    EXPECT_TRUE(iter.ReadString(&data));
    std::string big_string;
    EXPECT_TRUE(iter.ReadString(&big_string));
    EXPECT_EQ(kLongMessageStringNumBytes - 1, big_string.length());

    SendNextMessage();
    return true;
}

void TestChannelListener::OnChannelError()
{
    // There is a race when closing the channel so the last message may be lost.
    EXPECT_LE(messages_left_, 1);
    base::MessageLoop::current()->QuitWhenIdle();
}

void TestChannelListener::SendNextMessage()
{
    if (--messages_left_ <= 0)
        base::MessageLoop::current()->QuitWhenIdle();
    else
        SendOneMessage(sender_, "Foo");
}

}
